graph data structure algorithm in python